//给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。 
//
// 子数组是数组中元素的连续非空序列。 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [1,1,1], k = 2
//输出：2
// 
//
// 示例 2： 
//
// 
//输入：nums = [1,2,3], k = 3
//输出：2
// 
//
// 
//
// 提示： 
//
// 
// 1 <= nums.length <= 2 * 10⁴ 
// -1000 <= nums[i] <= 1000 
// -10⁷ <= k <= 10⁷ 
// 
//
// Related Topics 数组 哈希表 前缀和 👍 2417 👎 0


package leetcode.editor.cn;

import java.util.Arrays;
import java.util.HashMap;

class 和为K的子数组 {
    public static void main(String[] args) {
        Solution solution = new 和为K的子数组().new Solution();
        int[] nums = new int[]{1, 1, 1};
        int k = 2;
        System.out.println(solution.subarraySum(nums, k));
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        public int subarraySum(int[] nums, int k) {
            HashMap<Integer, Integer> map = new HashMap<>();
            int prev = 0, ans = 0;
            map.put(0 , 1);
            for (int num : nums) {
                prev += num;
                if (map.containsKey(prev - k)){
                    ans += map.get(prev - k);
                }
                map.put(prev, map.getOrDefault(prev, 0) + 1);
            }
            return ans;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}
